package socket;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Iterator;

public class Server {

    private ServerSocket serverSocket;
    //private PrintWriter[] allOut = {};
    private Collection<PrintWriter> allOut = new ArrayList<>();

    public Server(){
        try {
            System.out.println("正在启动服务端...");
            serverSocket = new ServerSocket(8088);
            System.out.println("服务端启动完毕");
        } catch (IOException e) {
            e.printStackTrace();
        }

    }

    public void start(){
        try {

            while (true) {
                System.out.println("等待客服端连接...");
                Socket socket = serverSocket.accept();
                System.out.println("一个客户端连接了");

                ClientHandler handler = new ClientHandler(socket);
                Thread t = new Thread(handler);
                t.start();

            }

        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        Server server = new Server();
        server.start();
    }

    private class ClientHandler implements Runnable{
        private Socket socket;
        private String host;   //记录当前对应客户端的IP地址

        public ClientHandler(Socket socket) {
            this.socket = socket;
            host = socket.getInetAddress().getHostAddress();
        }

        public void run(){
            PrintWriter pw = null;
            try {
            InputStream is = socket.getInputStream();
            InputStreamReader isr = new InputStreamReader(is, StandardCharsets.UTF_8);
            BufferedReader br = new BufferedReader(isr);

            //通过socket获取输出流对象给客服端发送消息
            OutputStream out = socket.getOutputStream();
            OutputStreamWriter osw = new OutputStreamWriter(out,StandardCharsets.UTF_8);
            BufferedWriter bw = new BufferedWriter(osw);
            pw = new PrintWriter(bw,true);

            //synchronized (Server.this){
            synchronized (allOut){
            allOut.add(pw);
            }
            sendMessage(host + "上线了，当前在线人数：" + allOut.size());


            String line ;
                while ((line = br.readLine()) != null){
                    sendMessage(host+"：" + line);

                }
            } catch (IOException e) {
            } finally {
                synchronized (Server.this) {
                    allOut.remove(pw);
                }
               sendMessage(host + "下线了，当前人数为：" + allOut.size());

                try {
                    socket.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }

        private void sendMessage(String message) {

            synchronized (allOut) {
                for (PrintWriter pw :
                        allOut) {
                    pw.println(message);
                    System.out.println(host + "：" + message);
                }
            }
        }
    }

}
